package com.hmall.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hmall.common.exception.BadRequestException;
import com.hmall.common.utils.UserContext;
import com.hmall.domain.dto.CartFormDTO;
import com.hmall.domain.po.Cart;
import com.hmall.domain.po.Item;
import com.hmall.domain.vo.CartVO;
import com.hmall.mapper.CartMapper;
import com.hmall.service.ICartService;
import com.hmall.service.IItemService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;

import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

@Service
@RequiredArgsConstructor
public class CartServiceImpl extends ServiceImpl<CartMapper, Cart> implements ICartService {

    private final IItemService itemService;

    @Override
    public void addItem2Cart(CartFormDTO cartFormDTO) {
        //TODO 如果商品已经添加过则购买数量添加即可；如果未添加过则新增一条购物车记录。一个用户最多放置购物车商品为10
        //  根据商品id查询商品是否存在；
        Long itemId = cartFormDTO.getItemId();
        Long userId = UserContext.getUser();
        //  如果存在则更新购物车数量；
        if (checkItemExist(itemId, userId)) {
            lambdaUpdate().setSql("num = num + 1")
                    .eq(Cart::getItemId, cartFormDTO.getItemId())
                    .eq(Cart::getUserId, UserContext.getUser())
                    .update();
        } else {
            //  如果不存在则新增一条购物车记录。
            if (getUserCartItemCount(userId) >= 10) {
                throw new BadRequestException("购物车商品数量不能超过10");
            }
            Cart cart = BeanUtil.copyProperties(cartFormDTO, Cart.class);
            cart.setUserId(userId);
            save(cart);
        }

    }


    /**
     * 获取当前用户购物车商品数量
     * @param userId
     * @return
     */
    private Long getUserCartItemCount(Long userId) {
        return lambdaQuery()
                .eq(Cart::getUserId, userId)
                .count();
    }

    /**
     * 判断商品是否已经添加到购物车
     *
     * @param itemId
     * @param userId
     * @return
     */
    private boolean checkItemExist(Long itemId, Long userId) {
        Long count = lambdaQuery().eq(Cart::getItemId, itemId).eq(Cart::getUserId, userId).count();
        return count > 0;
    }

    @Override
    public List<CartVO> queryMyCarts() {
        //TODO 查询当前登录用户的购物车列表；需要将Cart转换为CartVO；且CartVO中需要包含商品的最新价格、状态、库存等信息。
        // 查询当前登录用户的购物车列表
        List<Cart> cartList = lambdaQuery().eq(Cart::getUserId, UserContext.getUser()).list();
        if (CollUtil.isEmpty(cartList)) {
            return Collections.emptyList();
        }

        // 获取购物车中商品的id集合
        List<Long> itemIdList = cartList.stream().map(Cart::getItemId).collect(Collectors.toList());

        // 根据商品id集合批量查询商品
        List<Item> items = itemService.listByIds(itemIdList);
        Map<Long, Item> itemMap = items.stream().collect(Collectors.toMap(Item::getId, Function.identity()));

        // 转换Cart为CartVO并设置商品属性
        List<CartVO> cartVOList = cartList.stream()
                .map(cart -> {
                    CartVO cartVO = BeanUtil.copyProperties(cart, CartVO.class);
                    Item item = itemMap.get(cart.getItemId());
                    cartVO.setPrice(item.getPrice());
                    cartVO.setStatus(item.getStatus());
                    cartVO.setStock(item.getStock());
                    return cartVO;
                })
                .collect(Collectors.toList());

        return cartVOList;
    }

    @Override
    public void removeByItemIds(Collection<Long> itemIds) {
        // 1.构建删除条件，userId和itemId
        QueryWrapper<Cart> queryWrapper = new QueryWrapper<Cart>();
        queryWrapper.lambda().eq(Cart::getUserId, UserContext.getUser()).in(Cart::getItemId, itemIds);
        // 2.删除
        remove(queryWrapper);
    }
}
